package com.xiaoxiaoniu.airoleplay.repository;

import org.springframework.data.jpa.repository.Query;
import org.springframework.data.repository.query.Param;
import org.springframework.stereotype.Repository;
import com.xiaoxiaoniu.airoleplay.model.Message;
import org.springframework.data.jpa.repository.JpaRepository;

import java.util.List;

/**
 * 消息数据访问接口
 * 提供对消息实体的数据库操作方法
 */
@Repository
public interface MessageRepository extends JpaRepository<Message, Long> {

    /**
     * 根据会话ID查找消息，并按时间戳升序排列
     * 
     * @param conversationId 会话ID
     * @return 消息列表
     */
    List<Message> findByConversationIdOrderByTimestampAsc(String conversationId);

    /**
     * 根据会话ID查找最新的消息
     * 
     * @param conversationId 会话ID
     * @param pageable 分页参数
     * @return 消息列表
     */
    @Query("SELECT m FROM Message m WHERE m.conversationId = :conversationId ORDER BY m.timestamp DESC")
    List<Message> findLatestByConversationId(@Param("conversationId") String conversationId,
                                             org.springframework.data.domain.Pageable pageable);

    /**
     * 统计指定会话的消息数量
     * 
     * @param conversationId 会话ID
     * @return 消息数量
     */
    @Query("SELECT COUNT(m) FROM Message m WHERE m.conversationId = :conversationId")
    long countByConversationId(@Param("conversationId") String conversationId);

    /**
     * 根据角色查找消息，并按时间戳降序排列
     * 
     * @param role 角色名称
     * @param pageable 分页参数
     * @return 消息列表
     */
    @Query("SELECT m FROM Message m WHERE m.role = :role ORDER BY m.timestamp DESC")
    List<Message> findByRoleOrderByTimestampDesc(@Param("role") String role,
                                                 org.springframework.data.domain.Pageable pageable);
    
    /**
     * 查找孤立消息（没有关联会话的消息）
     * 
     * @param existingConversationIds 存在的会话ID列表
     * @return 孤立消息列表
     */
    @Query("SELECT m FROM Message m WHERE m.conversationId NOT IN :existingConversationIds")
    List<Message> findOrphanMessages(@Param("existingConversationIds") List<String> existingConversationIds);
}